package leetcode;

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Deque;
import java.util.List;

/**
 * @author niann
 * @description 第k个缺失的正整数
 * TODO https://leetcode.cn/problems/kth-missing-positive-number/description/
 * <p>
 * 给你一个 严格升序排列 的正整数数组 arr 和一个整数 k 。
 * 请你找到这个数组里第 k 个缺失的正整数。
 * @since 2024/8/31 下午12:36
 **/
public class LeetCode1539 {


    public int findKthPositive(int[] arr, int k) {
        Deque<Integer> deque = new ArrayDeque<>();
        for (int i = 0; i < arr.length + k; i++) {
            deque.add(i + 1);
        }
        for (int j : deque.stream().toList()) {
            for (int i : arr) {
                if (i == j) {
                    deque.remove(j);
                }
            }
        }
        return deque.stream().toList().get(k - 1);
    }

    /**
     * 解法2
     *
     * @param arr
     * @param k
     * @return
     */
    public int findKthPositive2(int[] arr, int k) {
        int j = 1;
        for (int i = 0; i < arr.length; ) {
            if (k == 0) {
                return j - 1;
            }
            if (arr[i] == j) {
                i++;
                j++;
            } else {
                j++;
                k--;
            }
        }
        while (k > 0) {
            j++;
            k--;
        }
        return j - 1;
    }
}
